#统计指定目录大小的函数
import os
def dir_size(dir):
    '''
    定义计算指定目录大小的函数
    '''
    sum = 0
    #判断指定目录是否为文件
    if os.path.isfile(dir):
        sum+=os.path.getsize(dir)
    #判断指定目录是否为文件夹
    if os.path.isdir(dir):
        #获取文件夹中的所有文件信息
        dlist = os.listdir(dir)
        for f in dlist:
            #为遍历的文件添加目录路径
            file = os.path.join(dir,f)
            if os.path.isfile(file):
                sum+=os.path.getsize(file)
            if os.path.isdir(file):
                sum+=dir_size(file) #递归调用统计目录大小函数
            
    return sum
while True:
    key_word = input("请输入指定目录的文件或文件夹，直接回车及退出\n")
    if key_word == "":
        break
    else:
        print("{}字节".format(dir_size(key_word)))